python train_clip_feature.py --aggregation=fedavg \
--epochs=10 \
--num_users=5 \
--device=cuda:2 \
--dataset=cifar10 \
--partition=dirichlet \
--beta=0.5 \
--batch_size=256 \
--model=cnn_clip
